package com.c2b.algorithm.leetcode.hot100;

import java.util.Stack;

/**
 * <a href="https://leetcode.cn/problems/convert-bst-to-greater-tree/">把二叉搜索树转换为累加树</a>
 * <p>给出二叉 搜索 树的根节点，该树的节点值各不相同，请你将其转换为累加树（Greater Sum Tree），使每个节点 node 的新值等于原树中大于或等于node.val的值之和。
 * <br><br>
 * 提醒一下，二叉搜索树满足下列约束条件：
 * <li>节点的左子树仅包含键 <b>小于</b> 节点键的节点。</li>
 * <li>节点的右子树仅包含键 <b>大于</b> 节点键的节点。</li>
 * <li>左右子树也必须是二叉搜索树。</li>
 * </p>
 * <pre>
 * 示例 1：
 *              4                                 30            30：8+7+6+5+4
 *           /    \                             /    \
 *          1      6                          36      21        21:8+7+6
 *        /  \    /  \      ==>              /  \    /  \
 *       0    2  5   7                     36   35  26   15     26:8+7+6+5
 *             \       \                         \        \     15:8+7
 *              3       8                         33       8    ...
 *      输入：[4,1,6,0,2,5,7,null,null,null,3,null,null,null,8]
 *      输出：[30,36,21,36,35,26,15,null,null,null,33,null,null,null,8]
 *
 * 示例 2：
 *      输入：root = [0,null,1]
 *      输出：[1,null,1]
 *
 *
 * 示例 3：
 *      输入：root = [1,0,2]
 *      输出：[3,3,2]
 *
 * 示例 4：
 *      输入：root = [3,2,4,1]
 *      输出：[7,9,4,10]
 * </pre>
 * <b>提示：
 * </b>
 * <li>树中的节点数介于 0 和 10^4 之间。</li>
 * <li>每个节点的值介于 -10^4 和 10^4 之间。</li>
 * <li>树中的所有值 互不相同 。</li>
 * <li>给定的树为二叉搜索树。</li>
 *
 * @author c2b
 * @since 2023/4/23 10:30
 */
public class Hot538ConvertBST_M {
    int sum = 0;

    public TreeNode convertBST(TreeNode root) {
        if (root != null) {
            convertBST(root.right);
            sum += root.val;
            root.val = sum;
            convertBST(root.left);
        }
        return root;
    }

    /**
     * 中序遍历(左根右)==》逆序：右根左
     */
    public void reverseMidOrderTraversal(TreeNode node) {
        if (node == null) {
            return;
        }
        Stack<TreeNode> stack = new Stack<>();
        TreeNode currNode = node;
        int initialSum = 0;
        while (currNode != null || !stack.isEmpty()) {
            while (currNode != null) {
                stack.push(currNode);
                currNode = currNode.right;
            }
            currNode = stack.pop();
            //
            initialSum += currNode.val;
            currNode.val = initialSum;
            currNode = currNode.left;
        }
    }

    public static void main(String[] args) {
        TreeNode treeNode = new TreeNode(4);
        treeNode.left = new TreeNode(1);
        treeNode.right = new TreeNode(6);
        treeNode.left.left = new TreeNode(0);
        treeNode.left.right = new TreeNode(2);
        treeNode.right.left = new TreeNode(5);
        treeNode.right.right = new TreeNode(7);
        treeNode.left.right.right = new TreeNode(3);
        treeNode.right.right.right = new TreeNode(8);
        Hot538ConvertBST_M hot538ConvertBST_m = new Hot538ConvertBST_M();
        TreeNode convertBST = hot538ConvertBST_m.convertBST(treeNode);
        System.out.println();
    }
}
